package com.aloda.utils;

public class TokensTreatUtils {

    /**
     * 预计算AI生成评分结果需要的tokens
     * @param questionContent 题目内容
     * @param resultNum 结果数量
     * @return 大概需要的tokens
     */
    public static Long calculateTokensGenerateResult(String questionContent, int resultNum) {
        return (questionContent.length() + resultNum * 10L);
    }

    /**
     * 预计算AI生成题目需要的tokens
     * @param questionNum 题目数量
     * @param optionNum 选项数量
     * @param appName 应用名称
     * @param appDesc 应用描述
     * @return 大概需要的tokens
     */
    public static Long calculateTokensGenerateQuestion(int questionNum, int optionNum, String appName, String appDesc) {
        return appName.length() + appDesc.length() + questionNum * 80L + optionNum * 80L;
    }
}
